package leetcode.editor.cn;
//给你一个非空数组，返回此数组中 第三大的数 。如果不存在，则返回数组中最大的数。 
//
// 
//
// 示例 1： 
//
// 
//输入：[3, 2, 1]
//输出：1
//解释：第三大的数是 1 。 
//
// 示例 2： 
//
// 
//输入：[1, 2]
//输出：2
//解释：第三大的数不存在, 所以返回最大的数 2 。
// 
//
// 示例 3： 
//
// 
//输入：[2, 2, 3, 1]
//输出：1
//解释：注意，要求返回第三大的数，是指在所有不同数字中排第三大的数。
//此例中存在两个值为 2 的数，它们都排第二。在所有不同数字中排第三大的数为 1 。 
//
// 
//
// 提示： 
//
// 
// 1 <= nums.length <= 10⁴ 
// -2³¹ <= nums[i] <= 2³¹ - 1 
// 
//
// 
//
// 进阶：你能设计一个时间复杂度 O(n) 的解决方案吗？ 
//
// Related Topics 数组 排序 👍 446 👎 0


import java.util.*;

//leetcode submit region begin(Prohibit modification and deletion)
class Solution209 {

    public static int thirdMax(int[] nums) {
        Integer first = null;
        Integer second = null;
        Integer third = null;

        for (int i = 0; i < nums.length; i++) {
            int num = nums[i];
            if ((first == null || second == null || third == null) || (num != first && num != second && num != third)) {
                if (first == null) {
                    first = num;
                } else if (second == null && num != first) {
                    if (num > first) {
                        second = first;
                        first = num;
                    } else {
                        second = num;
                    }
                } else if ((third == null && num != first && num != second) || (third != null && num > third)) {
                    if (num > first) {
                        third = second;
                        second = first;
                        first = num;
                    } else if (num > second) {
                        third = second;
                        second = num;
                    } else {
                        third = num;
                    }
                }
            }

        }
        return third == null ? first : third;
    }


    public int thirdMax4(int[] nums) {
        TreeSet<Integer> s = new TreeSet<Integer>();
        for (int num : nums) {
            s.add(num);
            if (s.size() > 3) {
                s.remove(s.first());
            }
        }
        return s.size() == 3 ? s.first() : s.last();
    }


    public int thirdMax2(int[] nums) {
        Arrays.sort(nums);
        int i = 1;
        int j = nums.length - 1;
        for (; j >= 0; j--) {
            if (i == 3 && nums[j] != nums[j + 1]) {
                return nums[j];
            }
            if (j == nums.length - 1 || nums[j] != nums[j + 1]) {
                i++;
            }
        }
        return nums[nums.length - 1];
    }


    public int thirdMax1(int[] nums) {
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        Set<Integer> set = new HashSet<>();
        for (int num : nums) {
            set.add(num);
        }
        for (int num : set) {
            pq.offer(num);
            if (pq.size() > 3) {
                pq.poll();
            }
        }
        if (set.size() < 3) {
            while (pq.size() > 1) pq.poll();
        }

        return pq.poll();
    }

    public static void main(String[] args) {
        System.out.println(thirdMax(new int[]{5, 2, 4, 1, 3, 6, 0}));
    }

}
//leetcode submit region end(Prohibit modification and deletion)
